import os


class DataReader:
    def __init__(self, data_dir):
        if data_dir[-1] != '/':
            data_dir += '/'
        self.data_dir_ = data_dir + "/data/"
        self.files_ = os.listdir(self.data_dir_)
        self.index_ = 0
        self.timestamp_file_ = open(data_dir + "/timestamps.txt")

    def has_more_sweep(self):
        return self.index_ < len(self.files_)

    def read_sweep(self):
        pass